package com.hxm.demo.test1;

/**
 * description: add a description
 *
 * @author huangxiaoming
 * @version 1.0.0
 * @date 2022/10/27 13:42:28
 */
public class TestFloodFill {

    /**
     * 733. 图像渲染
     *
     * @param args
     */
    public static void main(String[] args) {
        int[][] image = new int[][]{{1, 1, 1}, {1, 1, 0}, {1, 0, 1}};
        System.out.println(floodFill(image, 1, 1, 2));
    }

    /**
     * 有一幅以 m x n 的二维整数数组表示的图画 image ，其中 image[i][j] 表示该图画的像素值大小。
     * <p>
     * 你也被给予三个整数 sr ,  sc 和 newColor 。你应该从像素 image[sr][sc] 开始对图像进行 上色填充 。
     * <p>
     * 为了完成 上色工作 ，从初始像素开始，记录初始坐标的 上下左右四个方向上 像素值与初始坐标相同的相连像素点，接着再记录这四个方向上符合条件的像素点与他们对应 四个方向上 像素值与初始坐标相同的相连像素点，……，重复该过程。将所有有记录的像素点的颜色值改为 newColor 。
     * <p>
     * 最后返回 经过上色渲染后的图像 。
     *
     * @param image
     * @param sr
     * @param sc
     * @param color
     * @return
     */
    public static int[][] floodFill(int[][] image, int sr, int sc, int color) {
        int col = image.length;
        int row = image[0].length;
        int val = image[sr][sc];
        if (color == val) {
            return image;
        }
        boolean[][] booleans = new boolean[col][row];
        dp(image, booleans, sr, sc, color, val, col, row);
        return image;
    }

    public static void dp(int[][] image, boolean[][] booleans, int sr, int sc, int color, int val, int col, int row) {
        if (sr >= 0 && sc >= 0 && sr < col && sc < row && !booleans[sr][sc] && image[sr][sc] == val) {
            image[sr][sc] = color;
            booleans[sr][sc] = true;
            dp(image, booleans, sr - 1, sc, color, val, col, row);
            dp(image, booleans, sr + 1, sc, color, val, col, row);
            dp(image, booleans, sr, sc - 1, color, val, col, row);
            dp(image, booleans, sr, sc + 1, color, val, col, row);
        }
    }
}
